import sys
import os
from langchain.schema import HumanMessage
# 获取 user 目录的绝对路径
user_dir = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'customize')

# 将 user 目录添加到 sys.path 中
sys.path.append(user_dir)

from customize.get_ollama import GetOllama
r1 = GetOllama(model_type=0, model_name="deepseek-r1:14b")()
text = "解释一下鸡兔同笼问题。"
message = [HumanMessage(content=text)]
resp = r1.invoke(text)
print(resp)

